package com.example.demo;


import cn.hutool.extra.qrcode.QrCodeUtil;
import cn.hutool.extra.qrcode.QrConfig;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import org.springframework.core.io.DefaultResourceLoader;
import org.springframework.core.io.ResourceLoader;
import org.springframework.util.ResourceUtils;

import javax.imageio.ImageIO;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;


public class QRcodeImageFont {

    public static void main(String[] args) throws Exception {
        //在图片上生成二维码，并在图片上添加文字描述
        String url = "https://www.baidu.com";
        String savePath = "D://newLogo.png";
        GoodsInfoVO goodsInfoVO = new GoodsInfoVO();
        goodsInfoVO.setPicName("商品名称商品名称名称名称！！！！");
        QRcodeImageFont.drawString(url, savePath,goodsInfoVO);
    }

    /***
     * 在一张背景图上添加二维码
     */
    public static void drawString(String content, String savePath, GoodsInfoVO goodsInfoVO) throws Exception {
        BufferedImage image = addWater(content);//把二维码添加到图片上
        Graphics2D gd = image.createGraphics();
        // 3、设置对线段的锯齿状边缘处理
        gd.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
        // 5、设置水印文字颜色
        gd.setColor(Color.red);
        // 6、设置水印文字Font
        gd.setFont(new Font("宋体", Font.BOLD, 20));
        String name = goodsInfoVO.getPicName();
        // 8、第一参数->设置的内容，后面两个参数->文字在图片上的坐标位置(x,y)
        String picName = goodsInfoVO.getPicName();
        if(picName.length()>22) {
            picName = picName.substring(0, 21) + "...";
        }
            gd.drawString(name, 70, 50);
            gd.dispose();
        generateSaveFile(image, savePath);//保存文件到本地
        System.out.println("添加水印完成");
    }

    private static void generateSaveFile(BufferedImage buffImg, String savePath) {
        int temp = savePath.lastIndexOf(".") + 1;
        try {
            File outFile = new File(savePath);
            if (!outFile.exists()) {
                outFile.createNewFile();
            }
            ImageIO.write(buffImg, savePath.substring(temp), outFile);
            System.out.println("ImageIO write...");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /***
     * 在一张背景图上添加二维码
     */
    public static BufferedImage addWater(String content) throws Exception {
        // 读取原图片信息
        //得到文件
        File file = ResourceUtils.getFile("classpath:static/img/codeTwo.png");
        //文件转化为图片
        Image srcImg = ImageIO.read(file);
        //获取图片的宽
        int srcImgWidth = srcImg.getWidth(null);
        //获取图片的高
        int srcImgHeight = srcImg.getHeight(null);
//        //获取二维码的宽
//        int codeImgWidth = srcImg.getWidth(null)/5;//设置二维码为图片宽度为二维码图片的四分之一
//        //获取二维码的高
//        int codeImgHeight = srcImg.getHeight(null)/5;////设置二维码为图片高度为二维码图片的四分之一
//        int codeX = (srcImg.getWidth(null) - codeImgWidth) / 2; // 设置二维码图片的位置,这里令其居中
//        int codeY = (srcImg.getHeight(null) - codeImgHeight) / 2; // 设置二维码图片的位置,这里令其居中
//
//        int logoWidth = codeImgWidth/5; // 设置logo图片的宽度
//        int logoHegiht = codeImgHeight/5; // 设置logo图片的高度
        // 加水印
        BufferedImage bufImg = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = bufImg.createGraphics();
        g.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
        //使用工具类生成二维码 给二维码添加跳转路径并嵌套一个logo
        Image image = createQrCode(content, 200, 200);
        //将二维码绘到大图片上的位置。
        g.drawImage(image, 50, 180, null);
        //设置颜色。
        g.setColor(Color.WHITE);
        g.dispose();
        return bufImg;
    }

    private static BufferedImage createQrCode(String content, int width, int height) throws IOException {
        QrConfig config = new QrConfig(width, height);

        Image image = ImageIO.read(resourceLoader("classpath:static/img/logo.png"));
        config.setImg(image);
        config.setErrorCorrection(ErrorCorrectionLevel.H);
        return QrCodeUtil.generate(
                content,
                config);
    }

    public static InputStream resourceLoader(String fileFullPath) throws IOException {
        ResourceLoader resourceLoader = new DefaultResourceLoader();
        return resourceLoader.getResource(fileFullPath).getInputStream();
    }
}
